package ListNodeSecond.SwordOffer;

public class RemoveNthFromEnd {
    /**
     * 剑指 Offer II 021. 删除链表的倒数第 n 个结点
     * 给定一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。
     * <p>
     * 示例 1：
     * 输入：head = [1,2,3,4,5], n = 2
     * 输出：[1,2,3,5]
     * 示例 2：
     * <p>
     * 输入：head = [1], n = 1
     * 输出：[]A
     * 示例 3：
     * <p>
     * 输入：head = [1,2], n = 1
     * 输出：[1]
     *
     * @param head
     * @param n
     * @return
     */
    public ListNode removeNthFromEnd(ListNode head, int n) {
        if (head == null) {
            return null;
        }
        ListNode yummyNode = new ListNode(0, head);
        ListNode fastListNode = head;
        ListNode slowListNode = yummyNode;
        for (int i = 0; i < n; i++) {
            fastListNode = fastListNode.next;
        }
        while (fastListNode != null) {
            fastListNode = fastListNode.next;
            slowListNode = slowListNode.next;
        }
        slowListNode.next = slowListNode.next.next;
        return yummyNode.next;
    }
}
